def decorator(func):
    def wrapper(x):
        result = func(x)
        return result
    return wrapper


def func(x):
    return x**2
# 对函数使用装饰器 有两种方式
# 1
# @decorator
# def func(x):
#     return x**2
# 2
func = decorator(func)
print(func(2))